cmake_minimum_required(VERSION 3.5)

project(LinearRegression LANGUAGES CXX)

set(CMAKE_CXX_STANDARD 14)
set(CMAKE_CXX_STANDARD_REQUIRED ON)

find_package(Torch REQUIRED NO_MODULES PATHS "/opt/libtorch/share/cmake/Torch")
include_directories( ${TORCH_INCLUDE_DIRS} )

set(SRCS
    ./src/nets.cpp
    ./src/utils.cpp
    )

add_executable(${PROJECT_NAME} main.cpp ${SRCS})
target_include_directories(${PROJECT_NAME} PRIVATE ./inc )
target_link_libraries(${PROJECT_NAME} ${TORCH_LIBRARIES})
